﻿/*
给出一个完全二叉树，求出该树的节点个数。

说明：

完全二叉树的定义如下：在完全二叉树中，除了最底层节点可能没填满外，其余每层节点数都达到最大值，
并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层，则该层包含 1~ 2h 个节点。

示例:

输入:
	1
   / \
  2   3
 / \  /
4  5 6

输出: 6
*/

#include "AllInc.h"

struct TreeNode
{
	int val;
	TreeNode *left;
	TreeNode *right;
	TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

class Solution {
private:
	int height(TreeNode* root) 
	{
		if (root == NULL)
			return -1;
		while (root != NULL)
			return height(root->left) + 1;
		return 0;
	}
public:
	int countNodes(TreeNode* root) {
		int h = height(root);
		int nums = 0;
		while (root != NULL) {
			if (h - 1 == height(root->right)) {
				nums += 1 << h; //相当于pow(2,h - 1)
				root = root->right;
			}
			else {
				nums += 1 << h - 1;
				root = root->left;
			}
			h--;
		}
		return nums;
	}
};

//int main()
//{
//	TreeNode n1(1);
//	TreeNode n2(2);
//	TreeNode n3(3);
//	TreeNode n4(4);
//	n1.left = &n2;
//	n1.right = &n3;
//	n2.left = &n4;
//
//	Solution s;
//	auto result = s.countNodes(&n1);
//
//	return 0;
//}